#include "BlockingQueue.hpp"
#include "Task.hpp"
#include <iostream>
#include <unistd.h>
#include <ctime>

void *Consumer(void *args)
{
    BlockingQueue<Task> *bq = static_cast<BlockingQueue<Task> *>(args);
    while (true)
    {
        // 从阻塞队列获取数据
        // int data = 0;
        // bq->Pop(&data);
        // 处理数据
        // std::cout << "Consumer -> " << data << std::endl;

        // 获取数据
        Task t;
        bq->Pop(&t);

        // 处理数据
        t();
        std::cout << "Consumer -> " << t.result() << std::endl;
        std::cout << "======================================" << std::endl;
    }

    return nullptr;
}

void *Producer(void *args)
{
    BlockingQueue<Task> *bq = static_cast<BlockingQueue<Task> *>(args);
    srand(time(nullptr) ^ getpid());
    while (true)
    {
        sleep(1);
        // // 生产数据
        // int data = rand() % 10 + 1;
        // bq->Push(data);
        // // 处理数据
        // std::cout << "Producer -> " << data << std::endl;
        // sleep(1);

        // 准备数据
        int x = rand() % 10 + 1;
        usleep(100);
        int y = rand() % 10 + 1;

        // 生产数据
        Task t(x, y);
        bq->Push(t);

        // 处理数据
        std::cout << "Producer -> " <<t.debug() << std::endl;
        std::cout << "======================================" << std::endl;
    }

    return nullptr;
}

int main()
{
    // 创建一个阻塞队列
    BlockingQueue<Task> *bq = new BlockingQueue<Task>();
    // 创建两个线程
    pthread_t c1, c2, p1, p2, p3;
    pthread_create(&c1, nullptr, Consumer, bq); // 消费者
    pthread_create(&c2, nullptr, Consumer, bq); // 消费者
    pthread_create(&p1, nullptr, Producer, bq); // 生产者
    pthread_create(&p2, nullptr, Producer, bq); // 生产者
    pthread_create(&p3, nullptr, Producer, bq); // 生产者

    // 等待线程
    pthread_join(c1, nullptr);
    pthread_join(c2, nullptr);
    pthread_join(p1, nullptr);
    pthread_join(p2, nullptr);
    pthread_join(p3, nullptr);

    delete bq;
    return 0;
}
